matplotlib
numpy
scipy
ipdb
torch==0.4.1
torchvision==0.2.1
tqdm